import var_cnn_rnn_attention
import var_cnn_rnn
import var_cnn_rnn_attention_cnn
import config


def main():

    if config.attention_type == 'simple':
        model = var_cnn_rnn_attention.CRAttentionModel()
    elif config.attention_type == 'cnn':
        model = var_cnn_rnn_attention_cnn.CRAttentionCNNModel()
    else:
        model = var_cnn_rnn.CRModel()
    model.run()


if __name__ == '__main__':
    main()
